import torchline as tl
from torchline.config import CfgNode as CN

__all__ = [
    'add_config'
]

def add_config(cfg):
    cfg.model.lstm = CN()
    cfg.model.lstm.n_layer = 1
    cfg.model.lstm.in_dim = 1024
    cfg.model.lstm.hidden_dim = 512
    cfg.model.lstm.classes = 1024